using zip method of python